Type annotation for partial functions